package com.jt.interceptor;

import com.jt.pojo.User;
import com.jt.util.CookieUtil;
import com.jt.util.ObjectMapperUtil;
import com.jt.util.UserThreadLocal;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;
import org.springframework.web.servlet.HandlerInterceptor;
import redis.clients.jedis.JedisCluster;

import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
 * @author 张琪
 * @date 2020/12/17
 * @describe
 */
@Component
public class UserInterceptor implements HandlerInterceptor {

    public static final String JT_TICKET = "JT_TICKET";

    @Autowired
    private JedisCluster jedisCluster;

    /**
     * 业务描述:
     * 1. 判断用户是否登陆:    Cookie数据, redis中的数据
     * 2. 如果未登录 重定向到登陆页面, 否则直接进入
     *
     * @param request
     * @param response
     * @param handler
     * @return false 表示拦截  true 表示放行
     * @throws Exception
     */
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        /* 第一版代码 看起来较乱*/
/*
//        1. 校验cookie中是否有结果
        Cookie cookie = CookieUtil.getCookie(request, JT_TICKET);

//        2. 校验cookie是否有效
        if (cookie == null) {
            response.sendRedirect("/user/login.html");
            return false;
        }
        String ticket = cookie.getValue();
//        校验 对应cookie的值是否存在
        if (StringUtils.hasLength(ticket)) {
            if (jedisCluster.exists(ticket)) {
                return true;
            }
        } else {
//            如果取到的cookie中对应的值为空, 那么我们将删除cookie
            CookieUtil.addCookie(response, JT_TICKET);
            response.sendRedirect("/user.login.html");
        }

*/

        /*第二版 小问题,仍需修改*/
        /*String ticket = CookieUtil.getCookieValue(request, JT_TICKET);

        if (StringUtils.hasLength(ticket)) {
            if (jedisCluster.exists(ticket)) {
                return true;
            }
        } else {    // 存在问题: 当ticket为null时, 不需要删除cookie
            CookieUtil.addCookie(response, JT_TICKET);
        }
        response.sendRedirect("/user/login.html");*/

        Cookie cookie = CookieUtil.getCookie(request, JT_TICKET);

        if (cookie != null) {
            String ticket = cookie.getValue();
            if (StringUtils.hasLength(ticket)) {
                if (jedisCluster.exists(ticket)) {
//                    在这里获取用户信息
                    String json = jedisCluster.get(ticket);
                    User user = ObjectMapperUtil.toObject(json, User.class);
//                    方法1: 利用request对象, 存储数据进行传递
                    request.setAttribute("JT_USER", user);
//                    方法2: 利用ThreadLocal传递数据
                    UserThreadLocal.setUser(user);

                    return true;
                }
            }

            CookieUtil.addCookie(response, JT_TICKET);
        }


        response.sendRedirect("/user/login.html");
        return false;
    }

    //    为了防止内存的泄漏, 我们需要将多余的数据删除
    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
//        request的 数据删除
        request.removeAttribute("JT_USER");
//        ThreadLocal 的 数据删除
        UserThreadLocal.threadLocal.remove();
    }
}
